package org.common.utils.enums;
import cn.hutool.core.util.ClassUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.lang.reflect.Field;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 枚举工具
 * @author zhouzhibing
 * @date 2024/11/27
 */
public class EnumTool {

    private static final Logger log = LoggerFactory.getLogger(EnumTool.class);
    /**
     * 枚举实体列表<类型，<枚举ID/key，枚举对象>
     */
    private static final Map<Class<?> , Map<Integer, Object>> ENUM_INSTANCE_MAP = new ConcurrentHashMap<>();

    /**
     * 根据指定类型枚举和key，返回对应的枚举对象。
     * @param clzss 指定类型枚举
     * @param id 指定枚举中的key id
     * @return 返回枚举对象
     * @param <T> 枚举泛型类型
     */
    public static <T> T get(Class<T> clzss , int id) {
        Map<Integer, Object> enumMap = ENUM_INSTANCE_MAP.get(clzss);
        if(enumMap == null)
            throw new RuntimeException(clzss.getName() + ", not exist enum class.");

        T enumInstance = (T)enumMap.get(id);
        if(enumInstance == null)
            throw new RuntimeException(clzss.getName() + "." + id + ", not exist enum id.");

        return enumInstance;
    }

    /**
     * 初始化指定包下，使用@EnumKey注解的枚举。
     * @param packageName 包地址
     */
    public static void init(String packageName) {
        Set<Class<?>> classSet = ClassUtil.scanPackage(packageName);
        for (Class<?> clzss : classSet) {
            Field fieldKey = getEnumKeyField(clzss);
            if(fieldKey == null)
                continue;
            init(clzss , fieldKey);
        }
    }

    /**
     * 初始化指定类型枚举
     * @param clzss 指定类型枚举
     * @param <T> 枚举泛型类型
     */
    private static <T> void init(Class<T> clzss , Field fieldKey) {
        //不是该数值类型
        if (fieldKey.getType() != int.class) {
            throw new RuntimeException(clzss.getName() + "." + fieldKey.getName() + ", @EnumKey field type must is \"int\" .");
        }

        try {
            fieldKey.setAccessible(true);
            T[] enumConstants = clzss.getEnumConstants();
            for (T enumInstance : enumConstants) {
                int key = fieldKey.getInt(enumInstance);
                Map<Integer, Object> enumMap = ENUM_INSTANCE_MAP.computeIfAbsent(clzss, (k) -> new ConcurrentHashMap<>());
                if(enumMap.containsKey(key)) {
                    throw new RuntimeException(clzss.getName() + "." + fieldKey.getName() + ", exist duplicate key : \"" + key + "\".");
                }
                enumMap.put(key , enumInstance);
            }
        } catch (IllegalAccessException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 根据类型获取指定EnumKey的字段对象
     * @param clzss  类型
     * @return 获取指定EnumKey的字段对象
     * @param <T> 枚举泛型类型
     */
    private static <T> Field getEnumKeyField(Class<T> clzss) {
        Field[] fields = clzss.getDeclaredFields();
        for (Field field : fields) {
            EnumKey enumKey = field.getAnnotation(EnumKey.class);
            if(enumKey != null) {
                return field;
            }
        }
        return null;
    }
}
